package com.mybatispluslambda.utils;

import lombok.extern.slf4j.Slf4j;
import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.HanyuPinyinVCharType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

@Slf4j
public class ChinesePinyinUtils {

    /**
     * 将文字转为汉语拼音(原样输出,全拼)
     *
     * @param chineseName 汉语文字
     * @return 全拼
     */
    public static String getChineseWithAll(String chineseName) {
        char[] chrArray = chineseName.trim().toCharArray();
        StringBuilder chineseStr = new StringBuilder();
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        format.setCaseType(HanyuPinyinCaseType.LOWERCASE);// 输出拼音全部小写
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 不带声调
        format.setVCharType(HanyuPinyinVCharType.WITH_V);
        try {
            for (char chr : chrArray) {
                if (String.valueOf(chr).matches("[\u4e00-\u9fa5]+")) {// 如果字符是中文,则将中文转为汉语拼音
                    chineseStr.append(PinyinHelper.toHanyuPinyinStringArray(chr, format)[0]);
                } else {// 如果字符不是中文,则不转换
                    chineseStr.append(chr);
                }
            }
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            log.error("字符不能转成汉语拼音");
        }
        return chineseStr.toString();
    }

    /**
     * 将文字转为汉语拼音(过滤特殊字符只保留数字和字母,全拼)
     *
     * @param chineseName 汉语文字
     * @return 全拼
     */
    public static String getChineseWithNum(String chineseName) {
        char[] chrArray = chineseName.trim().toCharArray();
        StringBuilder chineseStr = new StringBuilder();
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        format.setCaseType(HanyuPinyinCaseType.LOWERCASE);   // 输出拼音全部大写
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 不带声调
        format.setVCharType(HanyuPinyinVCharType.WITH_V);
        try {
            for (char chr : chrArray) {
                String str = String.valueOf(chr);
                if (str.matches("[\u4e00-\u9fa5]+")) {// 如果字符是中文,则将中文转为汉语拼音,并取第一个字母
                    chineseStr.append(PinyinHelper.toHanyuPinyinStringArray(chr, format)[0]);
                } else if (str.matches("[0-9]+")) {// 如果字符是数字,取数字
                    chineseStr.append(chr);
                } else if (str.matches("[a-zA-Z]+")) {// 如果字符是字母,取字母
                    chineseStr.append(chr);
                } // 否则不转换
            }
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            log.error("字符不能转成汉语拼音");
        }
        return chineseStr.toString();
    }

    /**
     * 将文字转为汉语拼音(只转换汉语为拼音,全拼)
     *
     * @param chineseName 汉语文字
     * @return 全拼
     */
    public static String getChineseWithNone(String chineseName) {
        char[] chrArray = chineseName.trim().toCharArray();
        StringBuilder chineseStr = new StringBuilder();
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        format.setCaseType(HanyuPinyinCaseType.LOWERCASE);   // 输出拼音全部大写
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 不带声调
        format.setVCharType(HanyuPinyinVCharType.WITH_V);
        try {
            for (char chr : chrArray) {
                String str = String.valueOf(chr);
                if (str.matches("[\u4e00-\u9fa5]+")) {// 如果字符是中文,则将中文转为汉语拼音,并取第一个字母
                    chineseStr.append(PinyinHelper.toHanyuPinyinStringArray(chr, format)[0]);
                }
            }
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            log.error("字符不能转成汉语拼音");
        }
        return chineseStr.toString();
    }


    /**
     * 汉字首字母大写拼接(过滤特殊字符只保留数字和字母,全拼)
     *
     * @param chineseName 输入字符
     * @return 首字母大写
     */
    public static String getChineseFirstUpper(String chineseName) {
        return getFirstLetters(chineseName, HanyuPinyinCaseType.UPPERCASE);
    }

    /**
     * 汉字首字母小写拼接(接过滤特殊字符只保留数字和字母,全拼)
     *
     * @param chineseName 输入字符
     * @return 首字母小写
     */
    public static String getChineseFirstLower(String chineseName) {
        return getFirstLetters(chineseName, HanyuPinyinCaseType.LOWERCASE);
    }

    private static String getFirstLetters(String chineseName, HanyuPinyinCaseType caseType) {
        char[] charArray = chineseName.trim().toCharArray();
        StringBuilder chineseStr = new StringBuilder();
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        format.setCaseType(caseType);// 输出拼音全部大写
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 不带声调
        format.setVCharType(HanyuPinyinVCharType.WITH_V);
        try {
            for (char chr : charArray) {
                String str = String.valueOf(chr);
                if (str.matches("[\u4e00-\u9fa5]+")) {// 如果字符是中文,则将中文转为汉语拼音,并取第一个字母
                    chineseStr.append(PinyinHelper.toHanyuPinyinStringArray(chr, format)[0], 0, 1);
                } else if (str.matches("[0-9]+")) {// 如果字符是数字,取数字
                    chineseStr.append(chr);
                } else if (str.matches("[a-zA-Z]+")) {// 如果字符是字母,取字母
                    chineseStr.append(chr);
                } else {// 否则不转换
                    chineseStr.append(chr);//如果是标点符号的话，带着
                }
            }
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            log.error("字符不能转成汉语拼音");
        }
        return chineseStr.toString();
    }

    /**
     * 只取第一个汉字的第一个字符 (如果第一个字符不是汉字字母和数字, 则为空)
     *
     * @param chineseName 输入字符
     * @return 首字母(注意开头为汉字)
     */
    private static String getChineseFirstLetter(String chineseName) {
        char[] chrArray = chineseName.trim().toCharArray();
        String chineseStr = "";
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        format.setCaseType(HanyuPinyinCaseType.UPPERCASE);// 输出拼音全部大写
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 不带声调
        try {
            String str = String.valueOf(chrArray[0]);
            if (str.matches("[\u4e00-\u9fa5]+")) {// 如果字符是中文,则将中文转为汉语拼音,并取第一个字母
                chineseStr = PinyinHelper.toHanyuPinyinStringArray(chrArray[0], format)[0].substring(0, 1);
            } else if (str.matches("[0-9]+")) {   // 如果字符是数字,取数字
                chineseStr += chrArray[0];
            } else if (str.matches("[a-zA-Z]+")) {// 如果字符是字母,取字母
                chineseStr += chrArray[0];
            } // 否则不转换
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            log.error("字符不能转成汉语拼音");
        }
        return chineseStr;
    }
}